import pandas as pd
import numpy as np

# 1.使用 .loc[] 通过行名（索引）选择数据
data = {
    'A': [1, 2, 3],
    'B': [4, 5, 6]
}
df = pd.DataFrame(data, index=['row1', 'row2', 'row3'])

# 1.1 选择单行数据
# 选择行名为 'row1' 的数据
print(df.loc['row1'])

# 1.2 选择多个行名的数据
# 你可以通过传递一个包含多个行名的列表，来选择多个行的数据
# 选择 'row1' 和 'row3' 两行的数据
print(df.loc[['row1', 'row3']])

# 1.3 通过行名选择特定列的数据
# 选择 'row2' 行，'A' 列的数据
print(df.loc['row2', 'A'])


# 2.使用.iloc[]和位置选择
# 选择第一行
print("使用.iloc[]和位置选择")
print(df.iloc[0])
print(df.iloc[[0,1]])